package leetcode_800;

import java.util.Comparator;
import java.util.PriorityQueue;

/**
 *@author 周杨
 *KthSmallestPrimeFraction_786 给定一个排序的数组 问里面分子除以分母 第K的组合是什么
 *describe:用优先队列 AC 23%
 *2018年10月22日 上午11:59:57
 */
public class KthSmallestPrimeFraction_786 {
    public int[] kthSmallestPrimeFraction(int[] A, int K) {
    	int n = A.length;
        // 0: numerator idx, 1: denominator idx
        PriorityQueue<int[]> pq = new PriorityQueue<>(new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                int s1 = A[o1[0]] * A[o2[1]];
                int s2 = A[o2[0]] * A[o1[1]];
                return s1 - s2;
            }
        });
        for (int i = 0; i < n-1; i++) {
            pq.add(new int[]{i, n-1});
        }
        for (int i = 0; i < K-1; i++) {
            int[] pop = pq.remove();
            int ni = pop[0];
            int di = pop[1];
            if (pop[1] - 1 > pop[0]) {
                pop[1]--;
                pq.add(pop);
            }
        }
 
        int[] peek = pq.peek();
        return new int[]{A[peek[0]], A[peek[1]]};
    }

}
